import html2canvas from 'html2canvas'
import JsPDF from 'jspdf'

/**
 * @param  ele         要生成 pdf 的DOM元素（容器）
 * @param  pdfName     PDF文件生成后的文件名字
 * */
interface CanvasParams{dpi: number, useCORS: boolean}

function downloadPDF (ele: any, pdfName: string) {
    let eleW = ele.offsetWidth // 获得该容器的宽
    let eleH = ele.offsetHeight // 获得该容器的高

    let eleOffsetTop = ele.offsetTop // 获得该容器到文档顶部的距离
    let eleOffsetLeft = ele.offsetLeft // 获得该容器到文档最左的距离

    let canvas = document.createElement('canvas')
    let abs = 0

    let winIn = document.documentElement.clientWidth || document.body.clientWidth // 获得当前可视窗口的宽度（不包含滚动条）
    let winOut = window.innerWidth // 获得当前窗口的宽度（包含滚动条）

    if (winOut > winIn) {
        abs = (winOut - winIn) / 2 // 获得滚动条宽度的一半
    }

    canvas.width = eleW * 2 // 将画布宽&&高放大两倍
    canvas.height = eleH * 2

    const context: any = canvas.getContext('2d')

    context.scale(2, 2)

    context.translate(-eleOffsetLeft - abs, -eleOffsetTop)
    // 这里默认横向没有滚动条的情况，因为offset.left(),有无滚动条的时候存在差值，因此
    // translate的时候，要把这个差值去掉

    // html2canvas(element).then( (canvas)=>{ //报错
    // html2canvas(element[0]).then( (canvas)=>{
    const params:CanvasParams = {
        dpi: 300,
        // allowTaint: true,  // 允许 canvas 污染， allowTaint参数要去掉，否则是无法通过toDataURL导出canvas数据的
        useCORS: true // 允许canvas画布内 可以跨域请求外部链接图片, 允许跨域请求。
    }
    html2canvas(ele, params).then((canvas: any) => {
        var contentWidth = canvas.width
        var contentHeight = canvas.height
        // 一页pdf显示html页面生成的canvas高度;
        var pageHeight = contentWidth / 592.28 * 841.89
        // 未生成pdf的html页面高度
        var leftHeight = contentHeight
        // 页面偏移
        var position = 0
        // a4纸的尺寸[595.28,841.89]，html页面生成的canvas在pdf中图片的宽高
        var imgWidth = 595.28
        var imgHeight = 595.28 / contentWidth * contentHeight

        var pageData = canvas.toDataURL('image/jpeg', 1.0)
        var pdf = new JsPDF('', 'pt', 'a4')

        // 有两个高度需要区分，一个是html页面的实际高度，和生成pdf的页面高度(841.89)
        // 当内容未超过pdf一页显示的范围，无需分页
        if (leftHeight < pageHeight) {
            // 在pdf.addImage(pageData, 'JPEG', 左，上，宽度，高度)设置在pdf中显示；
            pdf.addImage(pageData, 'JPEG', 0, 0, imgWidth, imgHeight)
            // pdf.addImage(pageData, 'JPEG', 20, 40, imgWidth, imgHeight);
        } else { // 分页
            while (leftHeight > 0) {
                pdf.addImage(pageData, 'JPEG', 0, position, imgWidth, imgHeight)
                leftHeight -= pageHeight
                position -= 841.89
                // 避免添加空白页
                if (leftHeight > 0) {
                    pdf.addPage()
                }
            }
        }

        // 可动态生成
        pdf.save(pdfName)
    })
}

export default {
    downloadPDF
}
